#pragma once

// 声明CUDA函数，供C++调用
#ifdef __cplusplus
extern "C" {
#endif

    // 主机端函数，C++可以直接调用
    void cuda_add(int* a, int* b, int* c, int size);

#ifdef __cplusplus
}
#endif
